# ---------------------------------模型架构-----------------------------------------------
# 使用跨图交互
use_cross_interaction = True
# use_cross_interaction=False
crossNames = ["GMN", "cross_edge", "MGMN", "functionGMN"]
# cross_interaction_name=crossNames[0]
cross_interaction_name = crossNames[1]

# 使用异质信息
use_heterogeous = True
# use_heterogeous=False

# 图聚合方法
graphEmbedMethods = ["mean", "sum", "max"]
graph_embeddings_method = graphEmbedMethods[0]


# ---------------------------------超参数设置-----------------------------------------------
# 超参数设置
lr = 0.0001
epoch = 70  # default 100
laySize = 2
depth = 2
embSize = 128
margin = 0.7
wordsize = 8
tar_params_name = ""
tar_params_value = None
# batchSize不能改，代码好像存在问题
batchSize = 1

device = "cpu"
# device="cuda:0"

# ---------------------------------中间文件保存-----------------------------------------------
# 保存模型的函数的embedding
save_function_embedding = False
# save_function_embedding=True

# 显示test数据集的auc变换
show_test_auc = False

# ---------------------------------训练方法-----------------------------------------------
# 是否搜索参数
is_adjust_params = False

# 快速验证，用test和valid训练
quick_train = False
is_disassemble_experiment = False
disassemble_name = ["angr", "retdec", "radare2", "ida"]
# disassemble_tool = "angr"
# disassemble_tool = "retdec"
# disassemble_tool = "ida"
# disassemble_tool = "radare2"
